--extra-index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://pypi.nvidia.com
accelerate==0.25.0
build
colored
cuda-python # Do not override the custom version of cuda-python installed in the NGC PyTorch image.
diffusers==0.15.0
lark
mpi4py
numpy
onnx>=1.12.0
polygraphy
psutil
pynvml>=11.5.0
sentencepiece>=0.1.99
tensorrt==9.2.0.post12.dev5
torch<=2.2.0a
nvidia-ammo~=0.7.0; platform_machine=="x86_64"
transformers==4.36.1
wheel
optimum
evaluate
janus
